package com.linyaonan.leetcode.easy._643;

/**
 *
 * 给你一个由 n 个元素组成的整数数组 nums 和一个整数 k 。
 *
 * 请你找出平均数最大且 长度为 k 的连续子数组，并输出该最大平均数。
 *
 * 任何误差小于 10-5 的答案都将被视为正确答案。
 *
 *
 *
 * 示例 1：
 *
 * 输入：nums = [1,12,-5,-6,50,3], k = 4
 * 输出：12.75
 * 解释：最大平均数 (12-5-6+50)/4 = 51/4 = 12.75
 * 示例 2：
 *
 * 输入：nums = [5], k = 1
 * 输出：5.00000
 *
 *
 * 提示：
 *
 * n == nums.length
 * 1 <= k <= n <= 10^5
 * -10^4 <= nums[i] <= 10^4
 *
 * @author: Lin
 * @date: 2024/3/7
 */
public class MaximumAverageSubarrayI {

    public double findMaxAverage(int[] nums, int k) {

        long sum = 0L;
        long max = 0L;
        for (int i = 0; i < k; i++) {
            sum += nums[i];
        }
        max = sum;

        for (int i = k; i < nums.length; i++) {
            sum -= nums[i - k];
            sum += nums[i];
            max = Math.max(max, sum);
        }

        return (double) max / k;

    }

    public double findMaxAverage2(int[] nums, int k) {
        long max = Long.MIN_VALUE;
        long add = 0L;
        int count = 0;
        for (int i = 0; i < nums.length; i++) {
            add += nums[i];
            count++;
            if (count < k) {
                continue;
            }
            max = Math.max(max, add);
            count--;
            add -= nums[i - k + 1];
        }
        return (double) max / k;
    }

}
